/****************************************************************************
 * Copyright (C) 2009-2010 SciTouch LLC
 * 
 * This file is part of Indigo toolkit.
 * 
 * This file may be distributed and/or modified under the terms of the
 * GNU General Public License version 3 as published by the Free Software
 * Foundation and appearing in the file LICENSE.GPL included in the
 * packaging of this file.
 * 
 * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 * WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 ***************************************************************************/

#include "base_cpp/profiling.h"

#include <math.h>
#include "base_cpp/tlscont.h"
#include "base_cpp/output.h"
#include "base_cpp/reusable_obj_array.h"
#include "base_cpp/smart_output.h"

//
// _ProfilingTimer
//
_ProfilingTimer::_ProfilingTimer (int name_index)
{
   _name_index = name_index;
   _start_time = nanoClock();
}

_ProfilingTimer::~_ProfilingTimer ()
{
   stop();
}

qword _ProfilingTimer::stop ()
{
   if (_name_index == -1)
      return 0;

   ProfilingSystem &inst = ProfilingSystem::getInstance();
   _dt = nanoClock() - _start_time;
   inst.addTimer(_name_index, _dt);
   _name_index = -1;
   return _dt;
}

qword _ProfilingTimer::getTime () const
{
   if (_name_index == -1)
      return _dt;
   return nanoClock() - _start_time;
}

//
// Profiling functionality
//

OsLock _profiling_global_lock, _profiling_global_names_lock;
ObjArray< Array<char> > ProfilingSystem::_names;

TL_DECL(ProfilingSystem, _profiling_system);

ProfilingSystem& ProfilingSystem::getInstance ()
{
   TL_GET(ProfilingSystem, _profiling_system);
   return _profiling_system;
}

int ProfilingSystem::getNameIndex (const char *name, bool add_if_not_exists)
{
   OsLocker locker(_profiling_global_names_lock);

   for (int i = 0; i < _names.size(); i++)
      if (strcmp(_names[i].ptr(), name) == 0)
         return i;
   if (!add_if_not_exists)
      return -1;
   // Add new label
   Array<char> &name_record = _names.push();
   name_record.copy(name, strlen(name) + 1);
   return _names.size() - 1;
}

void ProfilingSystem::addTimer (int name_index, qword dt)
{
   OsLocker locker(_lock);

   _ensureRecordExistanceLocked(name_index);
   Record &rec = _records[name_index];
   rec.type = Record::TYPE_TIMER;
   rec.current.add(dt);
   rec.total.add(dt);
}

void ProfilingSystem::addCounter (int name_index, int value)
{
   OsLocker locker(_lock);

   _ensureRecordExistanceLocked(name_index);
   Record &rec = _records[name_index];
   rec.type = Record::TYPE_COUNTER;
   rec.current.add(value);
   rec.total.add(value);
}

void ProfilingSystem::reset (bool all)
{
   OsLocker locker(_lock);
   for (int i = 0; i < _records.size(); i++)
      _records[i].reset(all);
}

int ProfilingSystem::_recordsCmp (const int &idx1, const int &idx2, const void *context)
{
   return strcmp(_names[idx1].ptr(), _names[idx2].ptr());
}

void ProfilingSystem::getStatistics (Output &output, bool get_all)
{
   OsLocker locker(_lock);
   OsLocker names_locker(_profiling_global_names_lock);

   // Print formatted statistics
   while (_sorted_records.size() < _records.size())
      _sorted_records.push(_sorted_records.size());
   _sorted_records.qsort(_recordsCmp, this);

   // Find maximum name length
   int max_len = 0;
   for (int i = 0; i < _records.size(); i++)
   {
      if (!_hasLabelIndex(i))
         continue;

      if (_names[i].size() > max_len)
         max_len = _names[i].size();
   }

   SmartTableOutput table_output(output, true);

   table_output.setLineFormat("|c|5c|5c|");
   table_output.printHLine();
   table_output.printf("Name\tStatistics\t\t\t\t\tSession statistics\t\t\t\t\n");
   table_output.setLineFormat("|l|ccccc|ccccc|");
   table_output.printf("\ttotal\tcount\tavg.\tst.dev.\tmax\ttotal\tcount\tavg.\tst.dev.\tmax\n");
   table_output.printHLine();

   table_output.setLineFormat("|l|rrrrr|rrrrr|");

   for (int i = 0; i < _sorted_records.size(); i++)
   {
      int idx = _sorted_records[i];
      if (!_hasLabelIndex(idx))
         continue;
      Record &rec = _records[idx];
      if (!get_all && rec.current.count == 0)
         continue;

      table_output.printf("%s\t", _names[idx].ptr());

      if (rec.type == Record::TYPE_TIMER)
      {
         _printTimerData(rec.current, table_output);
         table_output.printf("\t"); 
         _printTimerData(rec.total, table_output);
         table_output.printf("\n"); 
      }
      else /* rec.type == Record::TYPE_COUNTER */
      {
         _printCounterData(rec.current, table_output);
         table_output.printf("\t"); 
         _printCounterData(rec.total, table_output);
         table_output.printf("\n"); 
      }
   }
   table_output.printHLine();

   table_output.flush();
}

void ProfilingSystem::_printTimerData (const Record::Data &data, Output &output)
{
   if (data.count == 0)
   {
      output.printf("-\t0\t\t\t"); 
      return;
   }
   float total_sec = nanoHowManySeconds(data.value);
   float avg_ms = nanoHowManySeconds(data.value / data.count) * 1000;
   float max_ms = nanoHowManySeconds(data.max_value) * 1000;

   double avg_value = (double)data.value / data.count;
   double sigma_sq = data.square_sum / data.count - avg_value * avg_value;
   float sigma_ms = nanoHowManySeconds((qword)sqrt(sigma_sq)) * 1000;

   output.printf("%0.2fs\t%0.0lf\t%0.1fms\t%0.1lfms\t%0.1fms",
      total_sec, (double)data.count, avg_ms, sigma_ms, max_ms); 
}

void ProfilingSystem::_printCounterData (const Record::Data &data, Output &output)
{
   if (data.count == 0)
   {
      output.printf("-\t0\t\t\t"); 
      return;
   }
   float avg_value = (float)data.value / data.count;

   // To avoid platform-specific code qwords were casted to doubles
   double sigma_sq = data.square_sum / data.count - avg_value * avg_value;
   output.printf("%0.0lf\t%0.0lf\t%0.1f\t%0.1lf\t%0.0lf",
      (double)data.value, (double)data.count, avg_value, sqrt(sigma_sq), (double)data.max_value); 
}

bool ProfilingSystem::_hasLabelIndex (int name_index)
{
   if (name_index >= _records.size())
      return false;
   return _records[name_index].total.count > 0;
}

bool ProfilingSystem::hasLabel (const char *name)
{
   int name_index = getNameIndex(name, false);
   if (name_index == -1)
      return false;
   return _hasLabelIndex(name_index);
}

void ProfilingSystem::_ensureRecordExistanceLocked (int name_index)
{
   while (_records.size() <= name_index)
      _records.push();
}

float ProfilingSystem::getLabelExecTime (const char *name)
{
   int idx = getNameIndex(name);
   OsLocker locker(_lock);
   _ensureRecordExistanceLocked(idx);
   return nanoHowManySeconds(_records[idx].current.value);
}

qword ProfilingSystem::getLabelCallCount (const char *name)
{
   int idx = getNameIndex(name);
   OsLocker locker(_lock);
   _ensureRecordExistanceLocked(idx);
   return _records[idx].current.count;
}

//
// ProfilingSystem::Record
//
void ProfilingSystem::Record::reset (bool all)
{
   current.reset();
   if (all)
      total.reset();
}

//
// ProfilingSystem::Record::Data
//

ProfilingSystem::Record::Data::Data ()
{
   reset();
}

void ProfilingSystem::Record::Data::reset () 
{ 
   count = value = max_value = 0;
   square_sum = 0; 
}

void ProfilingSystem::Record::Data::add (qword adding_value)
{
   count++;
   value += adding_value;
   max_value = __max(max_value, adding_value);

   double adding_value_dbl = (double)adding_value;
   square_sum += adding_value_dbl * adding_value_dbl;
}
